package com.sudo.common.task;

import java.util.HashMap;
import java.util.Map;

public class BaseTaskRegistry {
    final private static Map<String, Class<?>> beanClassMap = new HashMap<>();

    private BaseTaskRegistry() {
    }

    public static void setModuleClassMap(String key, Class<?> aClass){
        beanClassMap.put(key, aClass);
    }

    public static Map<String, Class<?>> getBeanClassMap(){
        return beanClassMap;
    }

    public static Class<?> getClassByBean(String beanName){
        return beanClassMap.get(beanName);
    }

}
